Updated the count update method
Added Update category method