Amazing Update: imp the controller function.