Update time function to use custom function instead
0.4.1