Small changes in js