Add roots attribute to the connection class
Remove unused code