Updated checkbox UI 2