Use np.concatenate rather than np.r_
Condense lines