Make the code work on the interpreter.

Noticed that the code was giving errors on trying to run on the interpreter. Thought that this might fix it.