Added stress tensor function, about to update velocity prediction