code changes

Signed-off-by: Connie Leung <cleung334@gmail.com>