Minor changes: Removed the print statements