Add any state change callback

I'm following the existing convention. So its not worth the change.