Update code, add function level type hints

Change-Id: Iaf158375cf41942ad6122b1a6a596483185e207e