Working on the last exam solution