write a python code