Add problem set 4