If statement added for question number four. Comments added for if statement.