Adding State class