assignment 2: added problem 01 D but a tests is failing for B and C