Sorry, had to PEP8ify the code :D