Modify the main.cpp into drawing a simple block