add class

Добавляем классы для определенных действий
Намного проще в эксплуатации и просмотре кода