Refactor to pycharm standards and make it look pretty.