Conform to PEP-8
Fix for #2