add test for new functionality