Fixed the type hint of  argument
Deleted unused variables