adding option for line fit