Used shared_task decorator for better compatibility