Add if statement for results