Fixed an error and updated some comments/prints