updated for load balancer and tests passing