Updated model declaration