WIP: fleshed out more in sample_k generation