Displaying checked/selected items as checked on edit page

Question

So i have this page where the user can can select items for two different categories (Cases and Parties). When a user creates a visit(schedule) they could select item or items f