clean up cky.py