Updating my_schedule to use mock schedule