Update for ass4