add tests for new functionality