problem set 2 bonus