Update to the `.on` event handler instead of `click`.