fix to main screen
and program UI