Modify comments
modify the type of K to uint8_t