finally got edit working on nested models