Added model for States in the frontend