Update test to draw square