could do with more tests (e.g. multiple arguments to function, different types of arguments, etc)