Got association working nicely