adding problem set that I did last night