Improves States example.