Removed erroneous import and added line before cell class for pep8