Fixed silly mistake working out the unit vector.