can now add class
a few changes

Now can add classes