Working on drawing cell mask and the tests