Improvements to the Custom Integration example.