Add some class in Draft