Add comment tests. Fix frontend bugs