Adding a first set of unit tests