clean code and add class
k
 ok