modify problem set 4 files.