Update edit add class

add class