Working on adding checkbox