added part of homework assignment