Minor changes, fixed a detail of the type hint