Small modification on code.py