added imports of linspace and log2