Started on basic unit test