Complete implementation

An image dataset is needed to run the code.