Update for assignment

As required