add code to controller