fixed bug in the solution.js

closes #4