cleanup needle imports and improve celery install