Update code.py
Merge pull request #4 from mmraka/my-new-branch

Update code.py