adding second part of hw1