Updated jQuery function