Add a function for better customization.