Update hw 4 - Make output clearer