End screen functionality implemented