Added unit test generation