hw 6 fixes