Python Input Array From User