I have put some if statements to display different results for different inputs.