Reformatting the code with Pycharm