Adding in prompt