Fix stats & end game screens