Changed example to numpy