updated after rubocop notes