Add address class in model