got poly fit working in 'class' for straight line case, need to implement python tests