update assignment controller and add count