Declare features in an initializer instead of model

To fix issues with code reloading and loading in general