Fixed example to use numpy explicitly in plotting