Improve type-checking for functions requiring time-series