fixed indentation etc with pycharm