minor fixes for prompts and handling non array returns