added part of hw2