Added some rubocop goodness