Imporving the UX with JQuery