Moved numpy include to appropriate point