add more tests to 04
update problem in section 4