Improve intstructions
Use context manager for connection socket
Prettify output
Clean whitespace