Improved UI, added some icons in the menu and model viewer