Custom classes work now. Need to fix securityCheck next.