Fixed code style to confirm to PEP 8