Adding support for new python modules... this might not work.