Added unit tests and corrected the answer model