now all with grid
add class heat
add implicit scheme
ready to pass