write tests for new features