[Fix] Refactor function.rs to handle empty input and returns

[Fix] Refactor function.rs to handle empty input and returns #140