Made my code bee-utiful with PEP8