Fixing up the js