More UI changes to print function