Add test for shuffle