added type annotation for map