Cleaning up the test.js file