fix main screen layout issues