Re-added code/surprise/surprise.py