new scrpt for hw4
new scrpt for hw4