Change psi to joint_feature based on pystruct change