Some tweaks

I will add other features like logging later.
Consider moving Rails specific feature from this gem.