Fix PEP8 and method