Add a unit tests