Update main.cpp

working on displacement