Add className method