Working on Controller error handling