You could perhaps start by adding type annotation. They are not perfect, but they help.