Add to Customer class