Added an if statement for my first prompt

Added an if statement for my first prompt