square class complete but ProbabilityMap class not yet using it