Added method chaining and override model load

to prevent unnecessary loading