added model back in