Add test for print