Ability to pass onBuild callback