End screen with code works