Clean up init.rb