Yes, I will change to enum. Too many if-else-ifs