Added some behavior in JS script