Add masks in problem interface