fix (hw 2,3,4)