Edited print statements slightly