Add missing annotations
Fix type annotation