start writing the python code