add more states. redo colors