Also accept a string as an argument

This would be a little extra exercise in API building