Adding code to get and sanitise user input