Update logical_cap.py