Change the output of the code so that all the degrees of freedom are outputted. This should be an easy fix (around 1 line of code according to Brian)