Made some changes for pycharm