Partial solution for square