Fix a minor bug to compute the biggest image
Update test case
Update image checking criteria