Problem two is finished