added missing state and models relations