Refactor datagrid stuff

| Q                    | A
| -------------------- | ---
| Bug fix?             | N
| New feature?         | N
| BC breaks?           | Y
| CI currently passes? | Y
| Tests pass?          | Y 
| Scenarios pass?      | Y
| Checkstyle issues?   | N
| Changelog updated?   | Y
| Fixed tickets        | -
| Doc PR               | -