Change checkbox wrapper div to use checkbox class