Updated tests for Comment class