Improved code

Improved code and adding images and HTML file for testing