Added tests and Comment class