HW4: add documentation