trying to remove onehot
working without onehot