create method on ViewModelFactory could more Kotlin-like

[henriquehorbovyi]

It would be better to use when expression to increase readability on this part of your code.

I'd like to open a PR for that 😄

[henriquehorbovyi]

Or this code will be removed after this PR? #3