final modifications for lab two