Assignment_5 updated