Adding time punch mocks
Modified for Time Punch