MAINT: Clean up unused imports and whitespace.
BUG: Reproducible testing. Closes #48