Change to be more pep8 compliant.

Move imports on one line, add two lines of empty space. General improvements.