Add a @model decorator for registering models