Got it working
Minor change in view