revise prob 4