Update assgnment2.sh