Add type annotations

To improve code quality and prevent subtle bugs this project should adopt type annotations.