Added import for scipy