problem set 4 make it works solved