Reformatted with pycharm