Assignment #5 update