Add focus class