Fixing function declaration
Fixing model