python accepting array