ENH: converted 2D lists to np arrays
MAINT: added testing lines