Updates to load test