Clean up screen function execution code