Add duration to first problem test function