fixes & clean up

Change-Id: I686870e063756c16d205fdc438adb6591a34461d