Added logic for screen dialog