Made code PEP8 conform