Got the modules code working