first shot at problem 4