Update main.cpp

This is where you'd give your code the width and height.