fixed bug in the solution.js

closes #4
Merge pull request #5 from bulkan/bugfix/whilst_solution

fixed bug in the solution.js