Added frontend logic of comments and improved server side for properly work