Made the inputs to _calc_rotations more general.
Progress on predict and fit