Implement window resize functionality.
Now squares don't stretch when the window is resized.