Simplify screen code