Made the output in the if loop make sense