Update cs hw 2