cleaning tests and print statements