fixing things
working on grid prediction