Switch to Mocha/Chai to test