Added screen handling class