After feedback from Pycharm. Perfect!