[WIP] Refactor screen module.