added images
structuring code