Scipy - the embarrassing way to code