improved acc to 0.56 by using factor variables instead of integer