Update ui.js

Little detail work before adding in div manager