World Models for Math Story Problems