Game screens/code fix up