Changed js a bit