Refactor: use specific data structure

This is more clear as we build a data structure specifically for
our use case.