remove prints
adjust test to pass .. understand later