Removed unnecessary space from class attributes, updated test for multiple classes