update controller
add new code