Changed the functionality by using dictionaries and adding images