Add in return statements to test.js