Added a way to specify a callback after the component has run