reworked part of end screen