Add needed type annotation. Make partial function not full function.