Update main.cpp

Remove divide by zero for data generation