Changing to pep 8 as per pycharm