add Problem C (not 100%)