add python solution for set game