Added missing import
Added support for multi-episodes

Fixes #8