Removed matplotlib and pandas import