Little Cleanup

Fixed some pep8 errors