Add assciation and validation