Removing unwanted print statements