Updated time slab heat driver
Added confusion shock problem