Fixing correct responses to display correctly