added tests for new features