Use graph model directly in images class