Add capture.py