MainScreen is a widget now