Add typinting