Add login states
Add model relations