Add setters and getters for models