Add classes and delete mistake