Major bug fixes

Fixed rendering bug after game over