Added a working version of KH, and updated the name used in the unittest