Added a square render function