My Python code