Adding className option