Changed 'add' to 'withClass' on TypedGraphModuleBuilder.