minor tweak to end screen