upated with rubocop