working on second problem set