Add custom checkbox directive