made changes to Assignment3