Add method to return marks