Updated User Interface to add check boxes