Created python file for code