adding back in prompt and fixing a/b mixup