Edit print function
Changed if condition