Update for Assignment1