probability map using the square class now