Adding mocha unit test