Make the module implementatino more generic to support distributions
other than Ubuntu in the future.

Adds distro as a new parameter.