Added unit tests as an exercise