Changed prompt to match new keys