adding work from cs 4590 hw2