Working WorldModel part 2