Toggle a 'focus' class on the btn