made comment prettier
added test for sample subset