problem2 work in progress