Add in test for the time_range function