Updated .module file to get everything to work