Added end screen functionality other than score