how to write a program on python