Finally got checkbox working