fix source format
add type annotation