Updated JS files to work