Remove print statements to allow python3 compatibility