Added sampling method using NN