updated : ASsignment 4 first draft