Removed numpy from import