work towards an end screen

work is commented