Add model registration and querying