Slightly cleaned up plus solved issue/simplified logic for level ending