Добавил .strip(),чтобы убрать пробелы в консоли.
Fixed error in import.