Adding Add Class Functionality