make it work with pycharm