Added variable and prompt on line 4