Add set-screen-wrapper! function