Updated to 4 states