Add comments controller and model classes; some refactor