Small but awesome optimizations to KL