Clean up screen function.