Allow test duration argument