Fixed bug and added ArrayList content #51

Forgot to insert line into Comment Model which caused the bug.