Added usage of registration of affected models in branch in different
kinds of models.

Signed-off-by: Jakob Brandsborg Ehmsen <jakobehmsen.github@gmail.com>