Addition of if statement to display results