Added hw 9 but doesn't work fully