Small bug fixes and clean up.
Added modular experiment runner class and cleaned up / unified some method signatures.