cleaned code
fixed rubocop errors