Updated pit states for rmax large grid world