corrected code to mask for future students