Cleaned up imports
Made @code_llvm_module return code instead of printing