Add states tests