Unit tests added for time