Code with python