+ add classes
+ fix bugs