Lab2 last fixes

Loads data from model without problem