Added functionality to print.py