Update code.py

Added description to file