Add railtie for rake task