Improve type annotations in the project

[einarf]

A couple of passes have been done already. Some temporary asserts was sprinkled around. These should ideally be removed in favor of proper typing.