Added python files for writing input, should be much more convenient!