Make the code work on the interpreter.
Merge pull request #3 from abhinav-upadhyay/code-fix

Make the code work on the interpreter.