Added missing method to mock
Added code folding todo