Added support for transition callbacks