removed print statements

and made test input data smaller