add optional className param