jQueriefied the code