Added load testing code