Update to jquery custom