how to write code in python