prob 2 work in progress