Code seems to be working fine

Did some changes in learning transformation test file to see whether the
output that we got from main code was correct or  not. The output is
matching and the images are exactly what I thought them to be. Thus, I
think code is working correctly right  now.