Add Possibility to Add Functionality to the State Transfer