Implement unit tests for new features.