Fixed issue with adding classes.