Updated KZ to 0.3.2

Updated KZ to 0.3.2, that has some slight API changes and package relocations.