Add extra requirement in setup.py