Adding an option for class name