fix arguments
add in logic for doing extrapolation with todo note