Added Module to class. Need to expose it to python.