Forgot to remove some print() calls