update the final version for hw 2