Worked on my algorithm, added the training image getter form