making UI work for main screen