Moved Polymock.py to models.py. Added convenience import to __init__.py