Including test cases for new functionality.