Adding unit test to test.js