Minor changes to jquery