Added classmethod for setting model config