MNT : cleaned up numpy usage
Merge pull request #2 from tacaswell/code_cleanup

MNT : cleaned up numpy usage