Rewrite the lib/setup.rb script to a Thor task