Added comments to dialog and image class