Added type annotation