Added a bit of code to js file