Chore/setup models