Improve the initial refactoring ```[tasklist] ### Tasks - [ ] #4 - [ ] #5 - [ ] #6 - [ ] #7 - [ ] #8 - [ ] #47 - [ ] #23 - [ ] https://github.com/rapidsai/wholegraph/issues/49 - [ ] #50 ```
Improve the initial refactoring