UPD: Updated model testing