Updated source code to PEP-8 Style Guide