Plex: Fixing boundary labels in create_unit_square