Rename code to code.py