Made code more better. 

Just a demo.
Update code.py
Merge pull request #4 from GoogleJump/demo

Made code more better. Don't you think?