Use enum for function and closure types in Rust

This container abstracts over our need to store closures somewhere, while allowing all "functions" to have the same type