Review kpm_funcs.py

This is a file that we haven't touched but are reusing from what was in the repo before. We should review it separately.