Add comment model
Add test for comments