Update code.py

Included brackets.