Remove formats (text styles) TinyMCE editor (Wordpress)

functions.php

<?php

// Modify TinyMCE editor to remove H1 and add some other options
function tiny_mce_remove_unused_formats_formatselect($init) {
  $init['block_formats'] = 'Paragraph=p;Heading 2=h2;Heading 3=h3;Heading 4=h4;Heading 5=h5;Heading 6=h6';
  return $init;
}

add_filter('tiny_mce_before_init', 'tiny_mce_remove_unused_formats_formatselect' );