Removed UI entities and moved logic to the screens.