clean up according to some rubocop stuff