the python code