updated for adding models