fixed imports and undefined method names