Reformated code with Pycharm