Added custom_message parameter