Added onSetup callback