Simplify prompt modification for Screen