added second part of hw