Updated assignment

Uploaded the wrong version previously