Update setup.py

Updated setup.py to include migrations