refactoring and improving kitty to support both simulation and quantile prediction