maybe just use numpy?