Added User model code