Keeping only wanted print statements