switched from numpy to pandas