Minor refactor of schema.rb