now working with regression