Fixed code to match PEP8 standart