Add module loading and function calling