PSet 3 solution