Updated path. Added option to pass in base class.

Fixes #9