Fixed a couple UI bugs with the checkbox