Task #200: Add if statements