Added doc comment type hint
Code cleanup