add class to support small screen