Change module to class