Completed part 1 of assignment