hw4part 2 with some content but to modify for sure