added endscreen, refactoring