add properly classes