Update cs106a assignment 1.2