Add states for every stage