add type hint / change @return