New version!

I have imported numpy, which makes it easier!