Fixed some rubocop things