add type annotation in printed code