Update main.cpp

Add temporary square generator (broken)