Add className option