Add type hints

Change-Id: I427eb4ab52f6ad80c993668f63326cde500e436b