Fixing up Contracts - ws.py