Making the stress classes work by fixing some of the errors related to tracelessness.
Also including the temperature as an input to the class!