PEP8 formatting fixes
Fixes bug where extract does not include warmup

Closes #101.