added end screen and generator

damn son