Add missing state definitions