Adding automated unit tests