Rename code to code.py

Tested adding .py extension.