Removed $comment parameter and used $this instead