Fixed and improved the custom example