My python code