Updated assignment

Updated assignment