Done first part of homework