Update code.py

Comments should be nice sentences.