Updated checkbox code