how to write python program