Update code to PEP 8