added more type annotations