updated stuff to work with pycharm