function annotation in python