removed unused import
use asserts to satisfy PEP8 checker... ugh