removed prints and fixed tests