minor bug with adding classes