Improve static safety of repo by using generic types and custom constructors

[davidmrdavid]

As per our discussion here: #184 (review)

Data such as the Entity batch array would be safer to use if registered as their own datatype. This way, we can prevent invalid operations across the codebase.

We should do a scan of the codebase and improve all the cases where custom types could be used to prevent invalid operation.