added changes to assgn1