need to add checkbox func