Add erd functionality to connection