Updating prompt_test to make it more debuggable in the future