First version of the world model + lookahead. It doesn't collapse anything but the results are not satisfactory yet.