Enhance AJAX functionality