Added tests, run with npm test