hw2-1b yet again