New python files + plotting script