how to write program in python