add basic functionality for end screen buttons