Added callback to the shuffle method