:white_check_mark: Add test to verify implementation