CSS focus does not change color after second clicking on button code example

Example 1: css keep focus color on div after click

$('button').on('click', function(){
    $('button').removeClass('selected');
    $(this).addClass('selected');
});

Example 2: css keep focus color on div after click

button.selected{
  color:red;
}