minor changes to print functionality