Fixes for end screen & word found