[patch] add types

[patch] add types #4