Create edited code.py