Change code to revealing module style