Writing a first round of unit tests