solved problem set of week 4