Fixing unnecessary print statements