Sherlock and Squares (python)