Small fixes to layout of end screen