Added class back to module template