add print to see how it works