Option to return array or models