How to write Python code?