Modified code to adhere to PEP8 guildelines