Added comments to main method for find_mask