added tests for new feature