Pass screen directly to wrapper function