adjusted mask to accommodate for new class