Changed up the JavaScript