Add display class