Fixed js a little bit