Updated to use mocha.

Added more tests and coverage report.