add 3rd part of homework