update to include state text labels