Refactor to work with any activemodel instance

Bump to 0.2.0