Update solution.js
Fix comments and refactoring
Try to change usage of babel
Add babel to dependencies