Add connection pooling code