Hack to handle images of different dimensions
Merge pull request #125 from vreddy18/master

Hack to handle images of different dimensions