Update shannon.py