Fix issue where squares are not drawn. Closes #5