Update to print functionality