Minor changes

Modify js to adapt to latest ui