Added checkbox functionality