code.py revised