Made code follow the flake8 standard