added numpy back in