Update kaptan/src/kaptan.py