Solution, but can't pass large set