Added a missing numpy include