Changed the setup.py to include Mock.