Add mocha unit testing back in again