You are right - the distribution is not equal largely because the dataset that I used had less pictures of harmless moles but I will aim to make it 50:50 in the next version