Add missing packages to setup.py
Add retry when setting up runner