Improve javascript

See how the javascript can be improved with or without jquery, if we use jQuery, make sure it works with zepto as well