Add skeleton code for Problem Set 4