changed to only one model