Missed an import.
Added some functions for different argument types.