MNT: Fixed print statements of demo behavior