Made checkbox work