Update main.cpp

add [size]