modify regression to accept argument