Problem B draft & tests
Problem B solution