UI update, fix bugs