Add ability to set marks