Flatten Function In Python