Added unit tests."