Added a few lines

Change-Id: Ibd02d0498a887f0d3ec57179dbe49310f48e5eeb