Add code to add/remove/get comments

Userstory: #22