Update main.cpp

Transformation matrix