Parameterize the init class. Allow package to be upgraded.

Just what the title says.
This is just nitpicking, but the order feels more logical if it's first the package and then the state to ensure.