diff --git a/register.py b/register.py
index 340a37325..4efdbaf17 100644
--- a/register.py
+++ b/register.py
@@ -2,13 +2,19 @@
 import os
 
 output = pypandoc.convert('README.md', 'rst')
-f = open('README.txt','w+')
-f.write(str(output.encode('utf-8')))
-f.close()
+with open('README.txt' 'w+') as f:
+    f.write(str(output.encode('utf-8')))
 
 readme_rst = open('./README.txt').read()
-replace = '.. figure:: https://uiux.s3.amazonaws.com/2016-logos/email-logo%402x.png\n   :alt: SendGrid Logo\n\n   SendGrid Logo\n'
-replacement = '|SendGrid Logo|\n\n.. |SendGrid Logo| image:: https://uiux.s3.amazonaws.com/2016-logos/email-logo%402x.png\n   :target: https://www.sendgrid.com'
-final_text = readme_rst.replace(replace,replacement)
+replace = '''
+            .. figure:: https://uiux.s3.amazonaws.com/2016-logos/email-logo
+            %402x.png\n   :alt: SendGrid Logo\n\n   SendGrid Logo\n
+          '''
+replacement = '''
+                |SendGrid Logo|\n\n.. |SendGrid Logo| image::
+                https://uiux.s3.amazonaws.com/2016-logos/email-logo%402x.png
+                \n   :target: https://www.sendgrid.com
+              '''
+final_text = readme_rst.replace(replace, replacement)
 with open('./README.txt', 'w') as f:
-  f.write(final_text)
+    f.write(final_text)