onehot as a module and main.py as a driver