improved a few print statements and comments