Add test for existing functionality.