Started setting up unit tests