Added python script to generate submission