Refactor screen loading.