Update jQuery method from .click() to .on()