cleaned up JITFunction codegen