Added changes to draw loop to try and get squares right

Loop