Add Celery support