Add class name as argument