Add checkbox logic