Add wheel module support