adding  parameter in get_connection function