Update main.cpp

whitening