Update mongo connection. Needs more work