Start writing unit tests