Add tests for new features and update README