Removed left over log statements
Added add_class and remove_class methods.