Made the print statements better