update end screen module to working version