Switched to mocha for unit tests