Fixed formatting according to Pycharm